from util import ContactProcessor

if __name__ == "__main__":

    npz_path = '/home/hlx/lsq/Squidiff/data/imputed_npz/G1/early_chr1cell_1_chr1_pad1_std1_rp0.5_sqrtvc.npz'
    chrom_sizes_path = '/home/hlx/lsq/Squidiff/chrom_size/mm9.chrom.sizes'
    chrom_name = 'chr1'
    chrom_size = 249250621
    resolution = 50000
    cool_path = '/home/hlx/lsq/Squidiff/utils/cell_1_raw_chr1.cool'


    contact_processor = ContactProcessor()
    contact_processor.npz_to_cooler_single_chrom(npz_path, chrom_name, chrom_size, resolution, cool_path)